package top;

/**
 * @author chenyw
 * @date 2022/7/8 20:01
 * @description     矩阵置零
 *      使用第一行和第一列作为标记所在的行或者列是否存在0，这样可以借读生出m+n的空间
 *      但是要先使用两个变量存储第一行和第一列开始时是否存在0
 */
public class Top73setZeroes {
    public void setZeroes(int[][] matrix) {
        int m = matrix.length;
        int n =matrix[0].length;
        boolean flagRow = false;
        boolean flagCol = false;
        for (int i = 0; i < m; i++) {
            if (matrix[i][0] == 0) {
                flagRow = true;
            }
        }
        for (int i = 0; i < n; i++) {
            if (matrix[0][i] == 0) {
                flagCol = true;
            }
        }
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (matrix[i][j] == 0) {
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }
        for (int i = 1; i < m; i++) {
            for (int j = 1; j < n; j++) {
                if (matrix[i][0] == 0 || matrix[0][j] == 0) {
                    matrix[i][j] = 0;
                }
            }
        }
        if (flagRow) {
            for (int i = 0; i < m; i++) {
                matrix[i][0] = 0;
            }
        }
        if (flagCol) {
            for (int i = 0; i < n; i++) {
                matrix[0][i] = 0;
            }
        }
    }
}
